package com.atguigu.flink.chapter02_DataStreamAPI.agg;

import com.atguigu.flink.pojo.WaterSensor;
import org.apache.flink.api.common.functions.MapFunction;
import org.apache.flink.api.common.functions.ReduceFunction;
import org.apache.flink.streaming.api.environment.StreamExecutionEnvironment;

/**
 * Created by Smexy on 2022/10/22
 *
 *      无分组(keyBy)，不聚合。
 *
 *      select
 *          sum()
 *      from xxx
 *      group by xx;
 *
 *      按照传感器的id，累加vc
 *
 *      ---------------------------
 *          高阶API，比较死板。不能指定reduce的累加的参数类型，及返回值类型。
 *              参数类型和返回值类型是固定的，就是流中数据的类型。
 *
 *      ------------------------
 *          滚动聚合(flink)：  来一条数据，立马聚合
 *          非滚动聚合： 等若干个数据来了后，一次性聚合
 *                      窗口中使用的多。
 *                      对某个窗口的数据，求topN
 *
 */
public class Demo16_Reduce
{
    public static void main(String[] args) {

        StreamExecutionEnvironment env = StreamExecutionEnvironment.getExecutionEnvironment();

        env
           .socketTextStream("hadoop103", 8888)
           .map(new MapFunction<String, WaterSensor>()
           {
               @Override
               public WaterSensor map(String value) throws Exception {
                   String[] data = value.split(",");
                   return new WaterSensor(
                       data[0],
                       Long.valueOf(data[1]),
                       Integer.valueOf(data[2])
                   );
               }
           })
           .keyBy(WaterSensor::getId)
           .reduce(new ReduceFunction<WaterSensor>()
           {
               @Override
               public WaterSensor reduce(WaterSensor value1, WaterSensor value2) throws Exception {

                   System.out.println("Demo16_Reduce.reduce");
                   //返回最新一条的其他属性+累加结果
                  // value2.setVc(value1.getVc() + value2.getVc());

                   //返回第一条数据的其他属性+累加结果
                   value1.setVc(value1.getVc() + value2.getVc());
                   return value1;
               }
           })
           .print();

        try {
                    env.execute();
                } catch (Exception e) {
                    e.printStackTrace();
                }
    }
}
